from scipy.signal import lti,step2,impulse2
import matplotlib.pyplot as plt
s1 = lti([50,0],[11,2,343,49])
s2 = lti([20,0,490],[11,2,343,49,0])

t1,y1=step2(s1)
t2,y2=step2(s2)
t11,y11=impulse2(s1)
t22,y22=impulse2(s2)

plt.plot(t1,y1,'r')
plt.title('H1 Step Response')
plt.show()

plt.plot(t2,y2,'b')
plt.title('H2 Step Response')
plt.show()

plt.plot(t11,y11,'r')
plt.title('H1 Impulse Response')
plt.show()

plt.plot(t22,y22,'b')
plt.title('H2 Impulse Response')
plt.show()
